shared source/wpf/src/host/shared/stringmap.hxx

   1 //------------------------------------------------------------------------
   2 //
   3 //  Copyright (c) Microsoft Corporation.  All rights reserved.
   4 //
   5 //  Description:
   6 //     Defines a list convenience class, for use with a small number of
   7 //     elements.
   8 //
   9 // History:
  10 //      2005/06/19 - [....]
  11 //          Created
  12 //      2007/09/20-[....]
  13 //          Ported Windows->DevDiv. See SourcesHistory.txt.
  14 //
  15 //------------------------------------------------------------------------
  16
  17 #pragma once
  18
  19 #ifndef STRINGMAP_HXX_INCLUDED
  20 #define STRINGMAP_HXX_INCLUDED
  21
  22 #include "string.hxx"
  23 #include "miscmacros.hxx"
  24
  25 template<class T>
  26 class CStringMap
  27 {
  28 public:
  29     class CStringMapNode
  30     {
  31     public:
  32         CStringMapNode(CStringMapNode* pNext, LPCWSTR key, T tValue)
  33         {
  34             m_pNext = pNext;
  35             SetKey(key);
  36             m_tValue = tValue;
  37         }
  38         STRING_PROP(Key);
  39         T GetValue() { return m_tValue; }
  40         CStringMapNode* GetNext() { return m_pNext; }
  41
  42     private:
  43         CString m_strKey;
  44         T m_tValue;
  45         CStringMapNode* m_pNext;
  46     };
  47
  48 public:
  49     CStringMap()
  50     {
  51         m_pRoot = NULL;
  52         m_dwCount = 0;
  53     }
  54
  55     ~CStringMap()
  56     {
  57         CStringMapNode* pNode = m_pRoot;
  58
  59         while (pNode)
  60         {
  61             CStringMapNode* pNext = pNode->GetNext();
  62             delete pNode;
  63             pNode = pNext;
  64         }
  65     }
  66
  67 public:
  68     HRESULT Add(LPCWSTR pwzKey, T tValue)
  69     {
  70         HRESULT hr = S_OK;
  71
  72         if (!pwzKey)
  73         {
  74             CKHR(E_INVALIDARG);
  75         }
  76
  77         m_pRoot = new CStringMapNode(m_pRoot, pwzKey, tValue);
  78         CK_ALLOC(m_pRoot);
  79         ++m_dwCount;
  80
  81     Cleanup:
  82         return hr;
  83     }
  84
  85     HRESULT Find(LPCWSTR pwzKey, T* ptValue)
  86     {
  87         HRESULT hr = S_OK;
  88
  89         CStringMapNode* pNode = m_pRoot;
  90
  91         while (pNode)
  92         {
  93             if (_wcsicmp(pwzKey, pNode->GetKey()) == 0)
  94             {
  95                 *ptValue = pNode->GetValue();
  96                 break;
  97             }
  98
  99             pNode = pNode->GetNext();
 100         }
 101
 102         if (!pNode)
 103         {
 104             CKHR(E_FAIL);
 105         }
 106
 107     Cleanup:
 108         return hr;
 109     }
 110
 111     CStringMapNode* GetRoot() { return m_pRoot; }
 112
 113     DWORD GetCount() { return m_dwCount; }
 114
 115 private:
 116     CStringMapNode* m_pRoot;
 117     DWORD m_dwCount;
 118 };
 119
 120 #endif //STRINGMAP_HXX_INCLUDED